package com.xyh.transaction.utils;

import com.alibaba.fastjson.JSON;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import java.util.Objects;
import java.util.Set;
import java.util.concurrent.TimeUnit;

@Component
public class RedisUtil {

    @Autowired
    private RedisTemplate<String, Object> redisTemplate;

    // ZSet 操作 - 向有序集合中添加元素
    public Boolean zsetAdd(String key, Object value, long score) {
        return redisTemplate.opsForZSet().add(key, value, score);
    }

    // ZSet 操作 - 获取有序集合中指定分数范围的元素
    public Set<Object> zsetRangeByScore(String key, long minScore, long maxScore) {
        return redisTemplate.opsForZSet().rangeByScore(key, minScore, maxScore);
    }

    // ZSet 操作 - 移除有序集合中的元素
    public Long zsetRemove(String key, Object... values) {
        return redisTemplate.opsForZSet().remove(key, values);
    }

    // ZSet 操作 - 获取有序集合中元素的排名
    public Long zsetRank(String key, Object value) {
        return redisTemplate.opsForZSet().rank(key, value);
    }

    // ZSet 操作 - 获取有序集合中元素的分数
    public Long zsetScore(String key, Object value) {
        return Objects.requireNonNull(redisTemplate.opsForZSet().score(key, value)).longValue();
    }


    // 获取键的过期时间
    public Long getExpire(String key) {
        return redisTemplate.getExpire(key);
    }

    // 获取键的过期时间，并指定时间单位
    public Long getExpire(String key, TimeUnit unit) {
        return redisTemplate.getExpire(key, unit);
    }

    // 设置缓存
    public <T> void set(String key, T value, long timeout, TimeUnit unit) {
        redisTemplate.opsForValue().set(key, value, timeout, unit);
    }

    // 获取缓存，使用 Class 对象以指定类型
    public <T> T get(String key, Class<T> clazz) {
        Object value = redisTemplate.opsForValue().get(key);
        return value != null ? JSON.parseObject(JSON.toJSONString(value), clazz) : null;
    }

    // 删除缓存
    public Boolean delete(String key) {
        return redisTemplate.delete(key);
    }

    // 判断缓存是否存在
    public Boolean hasKey(String key) {
        return redisTemplate.hasKey(key);
    }

    // 设置缓存的过期时间
    public Boolean expire(String key, long timeout, TimeUnit unit) {
        return redisTemplate.expire(key, timeout, unit);
    }

    // 自增操作
    public Long increment(String key, long delta) {
        return redisTemplate.opsForValue().increment(key, delta);
    }

    // 自减操作
    public Long decrement(String key, long delta) {
        return redisTemplate.opsForValue().increment(key, -delta);
    }

    // Hash 操作 - 设置值
    public void hashPut(String hashKey, String key, Object value) {
        redisTemplate.opsForHash().put(hashKey, key, value);
    }

    // Hash 操作 - 获取值
    public <T> T hashGet(String hashKey, String key, Class<T> clazz) {
        Object value = redisTemplate.opsForHash().get(hashKey, key);
        return value != null ? JSON.parseObject(JSON.toJSONString(value), clazz) : null;
    }

    // List 操作 - 向列表尾部添加元素
    public Long listRightPush(String key, Object value) {
        return redisTemplate.opsForList().rightPush(key, value);
    }

    // List 操作 - 获取列表范围内的元素
    public <T> T listRange(String key, long start, long end, Class<T> clazz) {
        return JSON.parseObject(JSON.toJSONString(redisTemplate.opsForList().range(key, start, end)), clazz);
    }
}
